tensor([[[0.3000, 0.3000],
         [0.3000, 0.3000]],

        [[0.4400, 0.4400],
         [0.4400, 0.4400]],

        [[0.2600, 0.2600],
         [0.2600, 0.2600]]], grad_fn=<RoundBackward1>)